//
//  UIButton+CountDown.m
//  Soha
//
//  Created by 谭飞 on 2018/9/2.
//  Copyright © 2018年 LB. All rights reserved.
//

#import "UIButton+CountDown.h"

@implementation UIButton (CountDown)

- (void)countDownWithTime:(NSInteger)timeLine withTitle:(NSString *)title andCountDownTitle:(nullable NSString *)subTitle countDownBlock:(void(^)(id obj))countDownBlock isEnabled:(BOOL)isEnabled {
    
    __block NSInteger timeout = timeLine - 1; // 倒计时时间
    
    dispatch_queue_t queue = dispatch_get_global_queue(DISPATCH_QUEUE_PRIORITY_DEFAULT, 0);
    
    dispatch_source_t _timer = dispatch_source_create(DISPATCH_SOURCE_TYPE_TIMER, 0, 0,queue);
    
    dispatch_source_set_timer(_timer,dispatch_walltime(NULL, 0),1.0*NSEC_PER_SEC, 0); // 每秒执行
    
    dispatch_source_set_event_handler(_timer, ^{
        
        if(timeout <= 0){ //倒计时结束，关闭
            dispatch_source_cancel(_timer);
            
            kDISPATCH_MAIN_THREAD(^{
                if (countDownBlock) {
                    countDownBlock(self);
                }
                //设置界面的按钮显示 根据自己需求设置
                self.userInteractionEnabled = YES;
                [self setTitle:title forState:UIControlStateNormal];
                self.titleLabel.font = WSFontMake(12);
            });
            
        } else {
            
            int seconds = timeout % 60;
        
            NSString *strTime = [NSString stringWithFormat:@"%.2d", seconds];
            if (seconds < 10) {
                strTime = [NSString stringWithFormat:@"%.1d", seconds];
            }
            
            kDISPATCH_MAIN_THREAD((^{
                
                //设置界面的按钮显示 根据自己需求设置
                if (subTitle) {
                    [self setTitle:[NSString stringWithFormat:@"(%@s)%@",strTime,subTitle] forState:UIControlStateNormal];
                } else {
                    [self setTitle:[NSString stringWithFormat:@"%@(%@s)",title,strTime] forState:UIControlStateNormal];
                }
                self.titleLabel.font = WSFontMake(12);
                self.userInteractionEnabled = isEnabled;
            }));
            
            timeout--;
        }
        
    });
    
    dispatch_resume(_timer);
}


@end
